package com.common.modelUtil;

import com.jfinal.kit.PathKit;
import com.jfinal.plugin.activerecord.dialect.MysqlDialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.druid.DruidPlugin;

import javax.sql.DataSource;

/**
 * Created by Administrator on 2019-12-30.
 */
public class GeneratorModel {
    //import com.common.model.*;
    static String jdbcUrl = "jdbc:mysql://120.76.28.44:3306/xj?characterEncoding=UTF-8&zeroDateTimeBehavior=convertToNull";
    static String user = "root";
    static String password = "ecuser01";

    public static DruidPlugin createDruidPlugin() {
        DruidPlugin druidPlugin = new DruidPlugin(jdbcUrl, user, password);
        return druidPlugin;
    }

    public static DataSource getDataSource() {
        DruidPlugin druidPlugin = createDruidPlugin();
        druidPlugin.start();
        return druidPlugin.getDataSource();
    }

    public static void main(String[] args) {
        // base model 所使用的包名
        String baseModelPackageName = "com.common.model.base";

        // base model 文件保存路径，注意该路径在实际开发时要指向  src/main/java/....
        // 此处仅为 demo 演示，指向了 src/test/java/....
        String baseModelOutputDir = PathKit.getWebRootPath() + "/src/main/java/com/common/model/base";

        // model 所使用的包名 (MappingKit 默认使用的包名)
        String modelPackageName = "com.common.model";
        // model 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
        String modelOutputDir = baseModelOutputDir + "/..";
        // MappingKit 文件保存路径 (MappingKit 与 DataDictionary 文件默认保存路径)
        String mappingKitPackageName = "com.common.utils";
        String mappingKitOutputDir = baseModelOutputDir + "/../../utils";


        // 创建生成器
        Generator generator = new Generator(getDataSource(), baseModelPackageName, baseModelOutputDir, modelPackageName, modelOutputDir);

        // 配置是否生成备注
//		generator.setGenerateRemarks(true);

        generator.setMetaBuilder(new  _MetaBuilder(getDataSource()));

        // 设置数据库方言
        generator.setDialect(new MysqlDialect());

        // 设置是否生成链式 setter 方法
        generator.setGenerateChainSetter(true);

        // 添加不需要生成的表名
//		generator.addExcludedTable("area","area_city","dicts","num","sys_no_seq","test_dr_cusprod_payoff","test_dr_cusprod_payon","ln_crm_cus_product_history");

        // 设置是否在 Model 中生成 dao 对象
        generator.setGenerateDaoInModel(true);

        // 设置是否生成字典文件
        generator.setGenerateDataDictionary(false);

        // 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
        generator.setRemovedTableNamePrefixes("xj_");

//		generator.setMappingKitOutputDir(mappingKitOutputDir);
//		generator.setMappingKitPackageName(mappingKitPackageName);
        generator.setMappingKitClassName("MappingKit");

        // 生成
        generator.generate();
    }
}
